// Platform.cpp
// (c) 2011, Charles Lechasseur
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
// 
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
// 
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.

#include "stdafx.h"
#include "Platform.h"

#include "Exceptions.h"

#include <ImageHlp.h>


namespace RG64
{
    //
    // Determines if a binary file is a 64-bit DLL or not.
    //
    // @param p_FilePath Path to the DLL file.
    // @return true if 64-bit DLL.
    //
    bool Is64Binary(const std::string& p_FilePath)
    {
        // Assume 32-bit.
        bool is64 = false;

        // Load image and make sure it is a PE image.
        LOADED_IMAGE* pImage = ::ImageLoad(p_FilePath.c_str(), 0);
        if (pImage == 0) {
            throw Win32Exception();
        } else {
            // Check machine flag to determine bitness.
            is64 = pImage->FileHeader->FileHeader.Machine == IMAGE_FILE_MACHINE_AMD64;

            // Don't forget to unload the image.
            ::ImageUnload(pImage);
        }

        return is64;
    }

    //
    // Determines whether we're running on a 64-bit OS or not.
    //
    // @return true if running on a 64-bit OS.
    //
    bool Is64BitOS()
    {
#ifdef _WIN64
        // If this is a 64-bit binary, we're obviously on a 64-bit OS.
        return true;
#else
        // Check if we're running in the WOW64 layer.
        BOOL isWow64 = FALSE;
        return ::IsWow64Process(::GetCurrentProcess(), &isWow64) != FALSE && isWow64 != FALSE;
#endif // _WIN64
    }

} // namespace RG64
